package com.starhub.application.rag.dto;

import lombok.Builder;
import lombok.Data;

/**
 * 检索请求dto
 * @version: V1.0
 * @author: liuhf
 * @time 2025-06-05
 */
@Builder
@Data
public class RetrieveDto {

    /**
     * 查询文本
     */
    private String queryText;
    
    /**
     * 查询元数据key,通过metadata检索时使用
     */
    private String queryMetaKey;

    /**
     * 返回结果数量
     */
    private Integer topK;
    
    /**
     * 最小相似度
     */
    private Double minScore;

    /**
     * 向量模型标识,来自知识库
     */
    private String vectorModel;
    
    /**
     * 向量维度，来自知识库
     */
    private int dimension;

    /**
     * 表id,来自知识库表
     */
    private String collectionId;

    /**
     * 表标识,来自知识库表
     */
    private String collectionIdentifier;

    /**
     * 是否进行重新排序
     */
    private boolean needRerank;

    /**
     * 使用的重排序模型
     */
    private String scoringModel;

}
